# model
model_name = 'conmh'
use_checkpoint = None
feature_size = 4096
hidden_size = 256
max_frames = 25
nbits = 16
transformer_type = 'small'

# dataset
dataset = 'hmdb'
workers = 12
batch_size = 128
mask_prob = 0.7

# train
seed = 1
num_epochs = 350
a = 1
temperature = 0.5
tau_plus = 0.05
train_num_sample = 3570

# test
test_batch_size = 128
test_num_sample = 3570 # test是test database的意思
query_num_sample = 1530 # query是test query

# optimizer
optimizer_name = 'AdamW'
schedule ="CosineAnnealingLR" #'StepLR'
lr = 5e-4
min_lr = 1e-5



# path
data_root = '/data2/lianniu/dataset/hmdb4/'
home_root = '/data2/lianniu/'

train_feat_path = [data_root + 'hmdb_train_feats.h5']

test_feat_path = [data_root + 'hmdb_test_feats.h5']
label_path = [data_root + 'hmdb_test_labels.mat']
test_feat_path = [data_root + 'hmdb_train_feats.h5'] # test是test database的意思
label_path = [data_root + 'hmdb_train_labels.mat']
query_feat_path = [data_root + 'hmdb_test_feats.h5'] # query是test query
query_label_path = [data_root + 'hmdb_test_labels.mat']

save_dir = home_root + 'saved_model/' + '_' + dataset
file_path = save_dir + '_' + str(nbits) + 'bit' + "test"

